package com.sz.admin.ai.controller.test;

import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

/**
 * @描述:
 * @Author: TangYuan
 * @Date: 2025/2/18 16:39
 */
@RestController
@RequestMapping("/ai/ollama/2")
public class OllamaTest2Controller {
    //
    // @Resource
    // private OllamaChatModel ollamaChatModel;
    // private ChatMemory chatMemory = new InMemoryChatMemory() {};
    //
    //
    // @GetMapping(value = "/steamCallWithHistory", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
    // public String chatStreamWithHistory(@RequestParam String msg, @RequestParam String sessionId) {
    //
    //     MessageChatMemoryAdvisor messageChatMemoryAdvisor = new MessageChatMemoryAdvisor(chatMemory, sessionId, 10);
    //
    //     ChatClient client = ChatClient.builder(ollamaChatModel).build();
    //     Prompt prompt = new Prompt(
    //             msg,
    //             OllamaOptions.builder()
    //                     .model("deepseek-r1:1.5b")
    //                     .temperature(0.7)
    //                     .build()
    //     );
    //     return client
    //             .prompt(prompt)
    //             .advisors(messageChatMemoryAdvisor)
    //             .call()
    //             .content();
    // }
    //
    // @GetMapping(value = "/steamCallWithHistory1", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
    // public Flux<String> chatStreamWithHistory1(@RequestParam String msg, @RequestParam String sessionId) {
    //
    //     Prompt prompt = new Prompt(
    //             msg,
    //             OllamaOptions.builder().model("deepseek-r1:1.5b")
    //                     .temperature(0.7)
    //                     .build()
    //     );
    //
    //     Flux<ChatResponse> stream = ollamaChatModel.stream(prompt);
    //     return stream.map(response -> {
    //         AssistantMessage output = response.getResult().getOutput();
    //         return output.getText();
    //     });
    // }
    //
    //
    //
    //

}
